﻿using EnvDTE;
using Microsoft.VisualStudio.Shell;
using System.ComponentModel.Design;
using System.Diagnostics;
using System.Globalization;
using System.IO;
using System.Runtime.InteropServices;
using System;

namespace VsExt.CommandFromHere
{
    /// <summary>
    /// This is the class that implements the package exposed by this assembly.
    ///
    /// The minimum requirement for a class to be considered a valid package for Visual Studio
    /// is to implement the IVsPackage interface and register itself with the shell.
    /// This package uses the helper classes defined inside the Managed Package Framework (MPF)
    /// to do it: it derives from the Package class that provides the implementation of the 
    /// IVsPackage interface and uses the registration attributes defined in the framework to 
    /// register itself and its components with the shell.
    /// </summary>
    // This attribute tells the PkgDef creation utility (CreatePkgDef.exe) that this class is
    // a package.
    // This attribute is used to register the information needed to show the this package
    // in the Help/About dialog of Visual Studio.
    // This attribute is needed to let the shell know that this package exposes some menus.
    [PackageRegistration(UseManagedResourcesOnly = true)]
    [InstalledProductRegistration("#110", "#112", "1.0", IconResourceID = 400)]
    [ProvideMenuResource("Menus.ctmenu", 1)]
    [Guid(GuidList.guidCommandFromHerePkgString)]
    public sealed class Parkware_CommandFromHerePackage : Package
    {
        private const string cmd = "cmd.exe";
        private const string variables32FileName = "vsvars32.bat";
        private const string VS100COMNTOOLS = "VS100COMNTOOLS";

        /// <summary>
        /// Default constructor of the package.
        /// Inside this method you can place any initialization code that does not require 
        /// any Visual Studio service because at this point the package object is created but 
        /// not sited yet inside Visual Studio environment. The place to do all the other 
        /// initialization is the Initialize method.
        /// </summary>
        public Parkware_CommandFromHerePackage()
        {
            Trace.WriteLine(string.Format(CultureInfo.CurrentCulture, "Entering constructor for: {0}", this.ToString()));
        }

        /// <summary>
        /// If the path contains any white space then we must wrap the path in quotes.
        /// </summary>
        /// <param name="path">The path to qualify</param>
        /// <returns></returns>
        public static string QualifyPath(string path)
        {
            return path.Contains(" ") ? string.Concat("\"", path, "\"") : path;
        }

        /// <summary>
        /// Initialization of the package; this method is called right after the package is sited, so this is the place
        /// where you can put all the initialization code that rely on services provided by VisualStudio.
        /// </summary>
        protected override void Initialize()
        {
            Trace.WriteLine(string.Format(CultureInfo.CurrentCulture, "Entering Initialize() of: {0}", this.ToString()));
            base.Initialize();

            // Add our command handlers for menu (commands must exist in the .vsct file)
            OleMenuCommandService mcs = GetService(typeof(IMenuCommandService)) as OleMenuCommandService;
            if (null != mcs)
            {
                // Create the command for the menu item.
                CommandID menuCommandID = new CommandID(GuidList.guidCommandFromHereCmdSetSolution, (int)PkgCmdIDList.cmdidCommandFromHereSolution);
                MenuCommand menuItem = new MenuCommand(MenuItemCallback, menuCommandID);
                mcs.AddCommand(menuItem);

                CommandID menuCommandProjectID = new CommandID(GuidList.guidCommandFromHereCmdSetProject, (int)PkgCmdIDList.cmdidCommandFromHereProject);
                MenuCommand menuItemProject = new MenuCommand(MenuItemCallback, menuCommandProjectID);
                mcs.AddCommand(menuItemProject);

                CommandID menuCommandFolderID = new CommandID(GuidList.guidCommandFromHereCmdSetFolder, (int)PkgCmdIDList.cmdidCommandFromHereFolder);
                MenuCommand menuItemFolder = new MenuCommand(MenuItemCallback, menuCommandFolderID);
                mcs.AddCommand(menuItemFolder);

                CommandID menuCommandItemID = new CommandID(GuidList.guidCommandFromHereCmdSetItem, (int)PkgCmdIDList.cmdidCommandFromHereItem);
                MenuCommand menuItemItem = new MenuCommand(MenuItemCallback, menuCommandItemID);
                mcs.AddCommand(menuItemItem);

                CommandID menuCommandCodeID = new CommandID(GuidList.guidCommandFromHereCmdSetCode, (int)PkgCmdIDList.cmdidCommandFromHereCode);
                MenuCommand menuItemCode = new MenuCommand(MenuItemCallback, menuCommandCodeID);
                mcs.AddCommand(menuItemCode);
            }
        }

        /// <summary>
        /// Gets the visual studio variables path.
        /// </summary>
        /// <returns></returns>
        private static string GetVisualStudioVarsPath()
        {
            string vars = string.Empty;

            string variablePath = Environment.GetEnvironmentVariable(VS100COMNTOOLS);

            if (variablePath != null)
            {
                vars = Path.Combine(variablePath, variables32FileName);
            }

            return vars;
        }

        /// <summary>
        /// This function is the callback used to execute a command when the a menu item is clicked.
        /// See the Initialize method to see how the menu item is associated to this function using
        /// the OleMenuCommandService service and the MenuCommand class.
        /// </summary>
        private void MenuItemCallback(object sender, EventArgs e)
        {
            DTE dte = (DTE)GetService(typeof(DTE));

            string directory = string.Empty;

            string visualStudioVariablesPath = GetVisualStudioVarsPath();

            if (dte.SelectedItems.Item(1).Project != null)
            {
                // project level
                directory = Path.GetDirectoryName(dte.SelectedItems.Item(1).Project.FullName);
            }
            else if (dte.SelectedItems.Item(1).ProjectItem != null)
            {
                // item or folder
                directory = Path.GetDirectoryName(dte.SelectedItems.Item(1).ProjectItem.get_FileNames(0));
            }
            else
            {
                // solution
                directory = Path.GetDirectoryName(dte.Solution.FullName);
            }

            Directory.SetCurrentDirectory(directory);

            if (File.Exists(visualStudioVariablesPath))
            {
                System.Diagnostics.Process.Start(cmd, string.Concat("/k \"" + QualifyPath(visualStudioVariablesPath)));
            }
        }
    }
}
